package com.LachesismStorm.lachesism.service.impl;

import com.LachesismStorm.lachesism.service.PubService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
import java.util.UUID;

/**
 * 图片处理
 * 常用文件的文件头如下：(以前六位为准)
 * JPEG (jpg)，文件头：FFD8FF
 * PNG (png)，文件头：89504E
 * GIF (gif)，文件头：474946
 * TIFF (tif)，文件头：49492A
 *
 * @Author: lachesism
 * @Date: 2022/11/07/17:29
 * @Description:
 */
@Service
@Slf4j
public class PubServiceImpl implements PubService {

    @Value("${file.dating.address}")
    private String datAddress;

    @Value("${url.pc}")
    private String webUrl;

    // 文件允许上传的最大大小，单位：字节（如 5MB）
    private static final long MAX_FILE_SIZE = 50 * 1024 * 1024;  // 50MB

    // 允许的文件扩展名
    private static final Set<String> ALLOWED_EXTENSIONS = new HashSet<>(Arrays.asList(".jpg", ".jpeg", ".png", ".gif", ".bmp"));

    @Override
    public String uploadPic(MultipartFile file) {
        if (file.isEmpty()) {
            log.error("上传的文件不能为空！");
            throw new FileUploadException("上传的文件不能为空！");
        }

        // 文件大小检查
        if (file.getSize() > MAX_FILE_SIZE) {
            log.error("文件大小超出限制，最大支持 {} MB", MAX_FILE_SIZE / 1024 / 1024);
            throw new FileUploadException("文件大小超出限制，最大支持 " + MAX_FILE_SIZE / 1024 / 1024 + "MB");
        }

        // 文件扩展名检查
        String originalFilename = file.getOriginalFilename();
        if (originalFilename == null || !isAllowedExtension(originalFilename)) {
            log.error("不支持的文件格式，支持的格式：{}", ALLOWED_EXTENSIONS);
            throw new FileUploadException("不支持的文件格式，支持的格式：" + ALLOWED_EXTENSIONS);
        }

        try {
            // 确保存储目录存在
            Path targetDir = Paths.get(datAddress);
            if (!Files.exists(targetDir)) {
                Files.createDirectories(targetDir);
            }
            // 生成唯一的文件名
            String extension = originalFilename.substring(originalFilename.lastIndexOf("."));
            String newFileName = UUID.randomUUID().toString().replace("-", "") + extension;
            // 目标文件路径
            Path destinationFilePath = targetDir.resolve(newFileName);
            file.transferTo(destinationFilePath.toFile());

            // 返回文件的访问 URL
            String fileUrl = webUrl + "image/" + newFileName;
            log.info("文件上传成功，文件URL: {}", fileUrl);
            return fileUrl;
        } catch (IOException e) {
            log.error("文件上传失败", e);
            throw new FileUploadException("文件上传失败", e);
        }
    }

    // 检查文件扩展名是否合法
    private boolean isAllowedExtension(String filename) {
        String extension = filename.substring(filename.lastIndexOf(".")).toLowerCase();
        return ALLOWED_EXTENSIONS.contains(extension);
    }

    // 自定义文件上传异常
    public static class FileUploadException extends RuntimeException {
        public FileUploadException(String message) {
            super(message);
        }

        public FileUploadException(String message, Throwable cause) {
            super(message, cause);
        }
    }
}
